#include "../server.hpp"

#include <fstream>
#include <sys/stat.h>
#include <regex>

#define DEFALT_TIMEOUT 10


// 状态信息
std::unordered_map<int, std::string> _statu_msg = {
    {100,  "Continue"},
    {101,  "Switching Protocol"},
    {102,  "Processing"},
    {103,  "Early Hints"},
    {200,  "OK"},
    {201,  "Created"},
    {202,  "Accepted"},
    {203,  "Non-Authoritative Information"},
    {204,  "No Content"},
    {205,  "Reset Content"},
    {206,  "Partial Content"},
    {207,  "Multi-Status"},
    {208,  "Already Reported"},
    {226,  "IM Used"},
    {300,  "Multiple Choice"},
    {301,  "Moved Permanently"},
    {302,  "Found"},
    {303,  "See Other"},
    {304,  "Not Modified"},
    {305,  "Use Proxy"},
    {306,  "unused"},
    {307,  "Temporary Redirect"},
    {308,  "Permanent Redirect"},
    {400,  "Bad Request"},
    {401,  "Unauthorized"},
    {402,  "Payment Required"},
    {403,  "Forbidden"},
    {404,  "Not Found"},
    {405,  "Method Not Allowed"},
    {406,  "Not Acceptable"},
    {407,  "Proxy Authentication Required"},
    {408,  "Request Timeout"},
    {409,  "Conflict"},
    {410,  "Gone"},
    {411,  "Length Required"},
    {412,  "Precondition Failed"},
    {413,  "Payload Too Large"},
    {414,  "URI Too Long"},
    {415,  "Unsupported Media Type"},
    {416,  "Range Not Satisfiable"},
    {417,  "Expectation Failed"},
    {418,  "I'm a teapot"},
    {421,  "Misdirected Request"},
    {422,  "Unprocessable Entity"},
    {423,  "Locked"},
    {424,  "Failed Dependency"},
    {425,  "Too Early"},
    {426,  "Upgrade Required"},
    {428,  "Precondition Required"},
    {429,  "Too Many Requests"},
    {431,  "Request Header Fields Too Large"},
    {451,  "Unavailable For Legal Reasons"},
    {501,  "Not Implemented"},
    {502,  "Bad Gateway"},
    {503,  "Service Unavailable"},
    {504,  "Gateway Timeout"},
    {505,  "HTTP Version Not Supported"},
    {506,  "Variant Also Negotiates"},
    {507,  "Insufficient Storage"},
    {508,  "Loop Detected"},
    {510,  "Not Extended"},
    {511,  "Network Authentication Required"}
};

// 文件后缀描述信息
std::unordered_map<std::string, std::string> _mime_msg = {
    {".aac",        "audio/aac"},
    {".abw",        "application/x-abiword"},
    {".arc",        "application/x-freearc"},
    {".avi",        "video/x-msvideo"},
    {".azw",        "application/vnd.amazon.ebook"},
    {".bin",        "application/octet-stream"},
    {".bmp",        "image/bmp"},
    {".bz",         "application/x-bzip"},
    {".bz2",        "application/x-bzip2"},
    {".csh",        "application/x-csh"},
    {".css",        "text/css"},
    {".csv",        "text/csv"},
    {".doc",        "application/msword"},
    {".docx",       "application/vnd.openxmlformats-officedocument.wordprocessingml.document"},
    {".eot",        "application/vnd.ms-fontobject"},
    {".epub",       "application/epub+zip"},
    {".gif",        "image/gif"},
    {".htm",        "text/html"},
    {".html",       "text/html"},
    {".ico",        "image/vnd.microsoft.icon"},
    {".ics",        "text/calendar"},
    {".jar",        "application/java-archive"},
    {".jpeg",       "image/jpeg"},
    {".jpg",        "image/jpeg"},
    {".js",         "text/javascript"},
    {".json",       "application/json"},
    {".jsonld",     "application/ld+json"},
    {".mid",        "audio/midi"},
    {".midi",       "audio/x-midi"},
    {".mjs",        "text/javascript"},
    {".mp3",        "audio/mpeg"},
    {".mpeg",       "video/mpeg"},
    {".mpkg",       "application/vnd.apple.installer+xml"},
    {".odp",        "application/vnd.oasis.opendocument.presentation"},
    {".ods",        "application/vnd.oasis.opendocument.spreadsheet"},
    {".odt",        "application/vnd.oasis.opendocument.text"},
    {".oga",        "audio/ogg"},
    {".ogv",        "video/ogg"},
    {".ogx",        "application/ogg"},
    {".otf",        "font/otf"},
    {".png",        "image/png"},
    {".pdf",        "application/pdf"},
    {".ppt",        "application/vnd.ms-powerpoint"},
    {".pptx",       "application/vnd.openxmlformats-officedocument.presentationml.presentation"},
    {".rar",        "application/x-rar-compressed"},
    {".rtf",        "application/rtf"},
    {".sh",         "application/x-sh"},
    {".svg",        "image/svg+xml"},
    {".swf",        "application/x-shockwave-flash"},
    {".tar",        "application/x-tar"},
    {".tif",        "image/tiff"},
    {".tiff",       "image/tiff"},
    {".ttf",        "font/ttf"},
    {".txt",        "text/plain"},
    {".vsd",        "application/vnd.visio"},
    {".wav",        "audio/wav"},
    {".weba",       "audio/webm"},
    {".webm",       "video/webm"},
    {".webp",       "image/webp"},
    {".woff",       "font/woff"},
    {".woff2",      "font/woff2"},
    {".xhtml",      "application/xhtml+xml"},
    {".xls",        "application/vnd.ms-excel"},
    {".xlsx",       "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"},
    {".xml",        "application/xml"},
    {".xul",        "application/vnd.mozilla.xul+xml"},
    {".zip",        "application/zip"},
    {".3gp",        "video/3gpp"},
    {".3g2",        "video/3gpp2"},
    {".7z",         "application/x-7z-compressed"}
};

class Util 
{
    public:
        //字符串分割函数,将src字符串按照sep字符进行分割，得到的各个字串放到arry中，最终返回字串的数量
        static ssize_t Split(const std::string& src,const std::string& sep,std::vector<std::string>* arry)
        {
            ssize_t offset = 0; // 偏移量
            // 注意偏移量的边界问题
            int n = src.size();
            while(offset < n)
            {
                ssize_t pos = src.find(sep,offset); // 从偏移量offset开始找，返回找到的sep的位置
                if(pos == std::string::npos) // 说明已经到头了
                {
                    // 将offset到pos部分插入到arry中，如果是空字符的话就不插入了
                    if(pos == n) break;
                    arry->push_back(src.substr(offset));
                    return arry->size();
                }
                if(offset == pos)// 说明是sep字符挨在一起的情况，比如sep=","时，出现 ,,,的情况
                {
                    offset++;
                    continue;
                }
                // 正常情况，就插入到arry中
                arry->push_back(src.substr(offset,pos - offset));
                offset = pos + sep.size();
            }

            return arry->size();
        }
        // 读取文件的所有内容，将其放到一个string中
        static bool ReadFile(const std::string& filename,std::string* buf)
        {
            std::ifstream ifs(filename,std::ios::binary); // 二进制方式打开
            if(ifs.is_open() == false)
            {
                ERR_LOG("Open %s File Failed! ",filename.c_str());
                return false;
            }
            ssize_t fsize = 0;
            ifs.seekg(0,ifs.end); // 先将读写指针移到末尾
            fsize = ifs.tellg(); // 获取当前读写位置相较于起始位置的偏移量，之前已经移动到末尾了，这样就能获取到文件的大小
            ifs.seekg(0,ifs.beg); // 再跳回开头的位置
            buf->resize(fsize); // 开辟对应的空间大小
            ifs.read(&(*buf)[0], fsize);
            if(ifs.good() == false) //good()表示ifs的上一次操作是否成功
            {
                // 读取失败了
                ERR_LOG("Read %s File Failed! ",filename.c_str());
                ifs.close();
                return false;
            }
            ifs.close();
            return true;
        }
        // 将string中的内容写入到文件中
        static bool WriteFile(const std::string& filename,const std::string& buf)
        {
            // std::ios::trunc表示每次打开文件会将文件清空写入
            std::ofstream ofs(filename,std::ios::binary | std::ios::trunc);
            if(ofs.is_open() == false)
            {
                ERR_LOG("Open %s File Failed! ",filename.c_str());
                return false;
            }
            ofs.write(buf.c_str(),buf.size());
            if(ofs.good() == false)
            {
                ERR_LOG("Write %s File Failed! ",filename.c_str());
                ofs.close();
                return false;
            }
            ofs.close();
            return true;
        }
        //URL编码，避免URL中资源路径与查询字符串中的特殊字符与HTTP请求中特殊字符产生歧义
        //编码格式：将特殊字符的ascii值，转换为两个16进制字符，前缀%   C++ -> C%2B%2B
        //  不编码的特殊字符： RFC3986文档规定 . - _ ~ 字母，数字属于绝对不编码字符
        //RFC3986文档规定，编码格式 %HH 
        //W3C标准中规定，查询字符串中的空格，需要编码为+， 解码则是+转空格
        //false 表示没有按照W3C标准
        static std::string URLEncode(const std::string url,bool convert_space_to_plus)
        {
            std::string res;
            for(auto& c : url)
            {
                //isalnum(c) 判断这个字符是否字母或者数字
                if(c == '.' || c == '-' || c == '_' || c == '~' || isalnum(c))
                {
                    res += c;
                    continue;
                }
                if(c == ' ' && convert_space_to_plus == true)
                {
                    res += '+';
                    continue;
                }
                //剩下的字符都是需要编码成为 %HH 格式
                char tmp[4] = {0};
                //snprintf 与 printf比较类似，都是格式化字符串，只不过一个是打印，一个是放到一块空间中,比如tmp中
                snprintf(tmp,4,"%%%02X",c);
                res += tmp;
            }
            return res;
        }
        // 十六进制转整形
        static char HEXTOI(char c)
        {
            if(c >= 0 && c <= '9') return c - '0';
            else if(c >= 'a' && c <= 'z') return c - 'a' + 10; // 注意十六进制，字母要+10
            else if(c >- 'A' && c <= 'Z') return c - 'A' + 10;
            else return -1; // 说明数据有误
        }
        static std::string URLDecode(const std::string url,bool convert_space_to_plus)
        {
            std::string res;
            int n = url.size();
            //遇到了%，则将紧随其后的2个字符，转换为数字，第一个数字左移4位，然后加上第二个数字  + -> 2b  %2b->2 << 4 + 11
            for(int i = 0; i < n; ++i)
            {
                if(url[i] == '+' && convert_space_to_plus == true)
                {
                    res += ' ';
                    continue;
                }
                if(url[i] == '%' && (i + 2) < n) // 注意防止越界
                {
                    char v1 = HEXTOI(url[i + 1]);
                    char v2 = HEXTOI(url[i + 2]);
                    char v = (v1 << 4) + v2;// 记得符号优先级，左移那里要加括号，或者也可以写成 v1 * 16 + v2
                    res += v;
                    i += 2; // 记得将i进行对应的偏移
                    continue;
                }
                //不需要转码的字符直接相加
                res += url[i];
            }

            return res;
        }
        //响应状态码的描述信息获取
        static std::string StatuDesc(int statu)
        {
            auto it = _statu_msg.find(statu);
            if(it != _statu_msg.end())
            {
                return it->second;
            }
            return "Unknow";
        }
        // 根据文件后缀名获取文件mime
        static std::string ExtMime(const std::string& filename)
        {
            // 先获取文件扩展名，注意a.b.txt这样的情况
            size_t pos = filename.find_last_of('.');
            if(pos == std::string::npos)
            {
                // 说明可能是个二进制文件
                return "application/octet-stream";
            }
            //根据扩展名，获取mime
            std::string ext = filename.substr(pos);
            auto it = _mime_msg.find(ext);
            if(it != _mime_msg.end())
            {
                return it->second;
            }
            return "application/octet-stream";
        }
        // 判断一个文件是否是目录
        static bool IsDirectory(const std::string& filename)
        {
            struct stat st;
            //获取 st结构体里面的st.st_mode的信息，然后用于判断是否是一个目录
            int ret = stat(filename.c_str(),&st);
            if(ret < 0)
            {
                return false;
            }
            // S_ISDIR是一个宏
            return S_ISDIR(st.st_mode);
        }
        // 判断一个文件是否是普通文件
        static bool IsRegular(const std::string& filename)
        {
            struct stat st;
            int ret = stat(filename.c_str(),&st);
            if(ret < 0)
            {
                return false;
            }
            return S_ISREG(st.st_mode);
        }
        //http请求的资源路径有效性判断
        // /index.html  --- 前边的/叫做相对根目录  映射的是某个服务器上的子目录
        // 想表达的意思就是，客户端只能请求相对根目录中的资源，其他地方的资源都不予理会
        // /../login, 这个路径中的..会让路径的查找跑到相对根目录之外，这是不合理的，不安全的
        static bool ValidPath(const std::string& path)
        {
            //思想：先按照/进行路径分割，根据有多少子目录，计算目录深度，有多少层，深度不能小于0
            std::vector<std::string> subdir;
            Split(path,"/",&subdir);
            int level = 0;
            for(auto& s : subdir)
            {
                if(s == "..") level--;
                else level++;
                if(level < 0) return false;// 说明已经不合法了
            }
            return true;
        }
};

class HttpRequest 
{
    public:
        std::string _method; // 请求方法
        std::string _path; //请求资源的路径
        std::string _version; // 协议版本
        std::string _body; // 请求正文
        std::smatch _matches; // 资源路径的正则提取数据
        std::unordered_map<std::string,std::string> _headers; // 头部字段信息
        std::unordered_map<std::string,std::string> _params; // 查询字段信息
    public:
        HttpRequest():_version("HTTP/1.1") {}
        // 重置参数
        void Reset() 
        {
            _method.clear();
            _path.clear();
            _version = "HTTP/1.1";
            _body.clear();
            std::smatch match;
            _matches.swap(match);
            _headers.clear();
            _params.clear();
        }
        // 插入头部字段
        void SetHeader(const std::string& key,const std::string& val)
        {
            _headers.insert({key,val}); // C++11用法，可以写成 std::make_pair(key,val)
        }
        // 判断是否存在指定头部字段
        bool HasHeader(const std::string& key) const
        {
            auto it = _headers.find(key);
            if(it != _headers.end())
            {
                return true;
            }
            return false;
        }
        // 获取指定头部字段的值
        std::string GetHeader(const std::string& key) const
        {
            auto it = _headers.find(key);
            if(it != _headers.end())
            {
                return it->second;
            }
            return "";
        }
        // 插入查询字符串
        void SetParam(const std::string& key,const std::string& val)
        {
            _params.insert({key,val});
        }
        // 判断是否有某个指定的查询字符串
        bool HasParam(const std::string& key) const
        {
            auto it = _params.find(key);
            if(it != _params.end())
            {
                return true;
            }
            return false;
        }
        // 获取指定的查询字符串
        std::string GetParam(const std::string& key) const
        {
            auto it = _params.find(key);
            if(it != _params.end())
            {
                return it->second;
            }
            return "";
        }
        // 获取正文长度
        ssize_t ContentLength() const 
        {
            // 格式类似这样 Content-Length: 1234\r\n，需要获取后面的数字
            if(HasHeader("Content-Length") == true)
            {
                return std::stol(GetHeader("Content-Length"));
            }
            return 0;
        }
        // 判断是否是短链接
        bool Close() const 
        {
            // 没有Connection字段，或者有Connection但是值是close，则都是短链接，否则就是长连接
            if(HasHeader("Connection") == true && GetHeader("Connection") == "keep-alive")
            {
                return false;
            }
            return true;
        }
};

class HttpResponse
{   
    public:
        int _statu; // 状态码
        bool _redirect_flag; // 是否要进行重定向，默认为false，不重定向
        std::string _body;
        std::string _redirect_url; // 重定向的URL
        std::unordered_map<std::string, std::string> _headers;
    public:
        HttpResponse(int statu = 200) : _statu(statu),_redirect_flag(false) {}
        void ReSet()
        {
            _statu = 200;
            _redirect_flag = false;
            _body.clear();
            _redirect_url.clear();
            _headers.clear();
        }
        // 插入头部字段
        void SetHeader(const std::string& key,const std::string& val)
        {
            _headers.insert({key,val}); // C++11用法，可以写成 std::make_pair(key,val)
        }
        // 判断是否存在指定头部字段
        bool HasHeader(const std::string& key) const
        {
            auto it = _headers.find(key);
            if(it != _headers.end())
            {
                return true;
            }
            return false;
        }
        // 获取指定头部字段的值
        std::string GetHeader(const std::string& key) const
        {
            auto it = _headers.find(key);
            if(it != _headers.end())
            {
                return it->second;
            }
            return "";
        }
        // 设置正文内容，和正文类型
        void SetContent(const std::string& body,const std::string& type = "text/html")
        {
            _body = body;
            _headers.insert({"Content-Type",type});
        }
        // 启动重定向
        void SetRedirect(const std::string& url,int statu = 302)
        {
            _statu = statu;
            _redirect_flag = true;
            _redirect_url = url;
        }
        // 判断是否是短链接
        bool Close() const 
        {
            // 没有Connection字段，或者有Connection但是值是close，则都是短链接，否则就是长连接
            if(HasHeader("Connection") == true && GetHeader("Connection") == "keep-alive")
            {
                return false;
            }
            return true;
        }
};

// 对一个连接进行业务处理时的不同状态
typedef enum
{
    RECV_HTTP_ERROR,
    RECV_HTTP_LINE,
    RECV_HTTP_HEAD,
    RECV_HTTP_BODY,
    RECV_HTTP_OVER
}HttpRecvStatu;

// 一行信息的最大值
#define MAX_LINE 8192
// 对一个请求进行业务处理，并能保存上下文的进度
class HttpContext 
{
    private:
        int _resp_statu; // 响应状态码
        HttpRecvStatu _recv_statu; // 当前请求解析的阶段（状态）
        HttpRequest _request; // 将解析好的内容填充到这个请求结构体中
    private:
        // 解析出错后，设置对应的状态信息
        void ParseERROR(int statu = 400)
        {
            _recv_statu = RECV_HTTP_ERROR;
            _resp_statu = statu; // 400 ->BAD REQUEST 414 -> URI TOO LONG
        }
        // 解析首行信息
        bool ParseHttpLine(const std::string& line)
        {
            std::smatch matches;
            // std::regex::icase 无视大小写，增加兼容性
            std::regex e("(GET|HEAD|POST|PUT|DELETE) ([^?]*)(?:\\?(.*))? (HTTP/1\\.[01])(?:\n|\r\n)?", std::regex::icase);
            bool ret = std::regex_match(line, matches, e);
            if(ret == false)
            {
                // 设置出错信息
                ParseERROR(400);// 400 ->BAD REQUEST
                return false;
            }
            // 示例解析后的格式：（存放在了matchs中，0 1 2 3...对应的matchs的下标，可以用这个下标来直接提取匹配后的结果）
            //0 : GET /nihao/login?user=xiaoming&pass=123123 HTTP/1.1
            //1 : GET
            //2 : /nihao/login
            //3 : user=xiaoming&pass=123123
            //4 : HTTP/1.1
            // 请求方法的获取
            _request._method = matches[1];
            //将请求方法的字符统一转换称为大写
            std::transform(_request._method.begin(),_request._method.end(),_request._method.end(),::toupper);
            // 资源路径的获取，需要进行URL解码操作，这里不需要 + 转换空格
            _request._path = Util::URLDecode(matches[2],false);
            // 协议版本的获取
            _request._version = matches[4];
            // 查询字符串的获取和处理
            std::vector<std::string> query_string_arry;
            std::string query_string = matches[3];
            // 查询字符串的格式 key=val&key=val....., 先以 & 符号进行分割，得到各个字串,然后放到vector中
            Util::Split(query_string ,"&",&query_string_arry);
            //针对各个字串，以 = 符号进行分割，得到key 和val， 得到之后也需要进行URL解码
            for(auto& s : query_string_arry)
            {
                size_t pos= s.find('=');
                if(pos == std::string::npos)
                {
                    // 说明不符合格式
                    ParseERROR(400); // BAD REQUEST
                    return false;
                }
                // 这里要 + 转空格
                std::string key = Util::URLDecode(s.substr(0,pos),true);
                std::string val = Util::URLDecode(s.substr(pos + 1),true);
                _request.SetParam(key,val);
            }
            return true;
        }
        // 获取/接收首行信息,并且进行解析
        bool RecvHttpLine(Buffer& buf)
        {     
            if(_recv_statu != RECV_HTTP_LINE) return false; // 不是当前的处理状态直接出错返回
            // 1.获取一行数据，并带有换行符
            std::string line = buf.GetLineAndPop();
            // 如果这一行的数据过大也不行，不予接收
            if(line.size() > MAX_LINE)
            {
                _recv_statu = RECV_HTTP_ERROR;
                _resp_statu = 414; // URI TOO LONG
                return false;
            }
            if(line.size() == 0)
            {
                // 虽然没有一行的数据，但是如果缓冲区里的数据已经很大了，但是还是不足一行，也不行
                if(buf.ReadAbleSize() > MAX_LINE)
                {
                    ParseERROR(414);; // URI TOO LONG
                    return false;
                }
                // 说明数据量合理，但是不足一行，那么下次再处理
                return true;
            }
            bool ret = ParseHttpLine(line);
            if(ret == false)
            {
                return false;
            }

            // 到这里说明首行处理完毕，进入下一阶段
            _recv_statu = RECV_HTTP_HEAD;
            return true;
        }
        // 获取并解析头部信息
        bool RecvHttpHead(Buffer& buf)
        {
            if(_recv_statu != RECV_HTTP_HEAD) return false;
            // 一行一行取出数据，直到遇到空行为止， 头部的格式 key: val\r\nkey: val\r\n....
            // 所以要用while循环来读，直到读到一行的数据就只有一个 换行符位置
            while(1)
            {
                std::string line = buf.GetLineAndPop();
                // 同样需要考虑，是否一行的数据过大
                // 如果这一行的数据过大也不行，不予接收
                if(line.size() > MAX_LINE)
                {
                    ParseERROR(414); // URI TOO LONG
                    return false;
                }
                if(line.size() == 0)
                {
                    // 虽然没有一行的数据，但是如果缓冲区里的数据已经很大了，但是还是不足一行，也不行
                    if(buf.ReadAbleSize() > MAX_LINE)
                    {
                        ParseERROR(414); // URI TOO LONG
                        return false;
                    }
                    // 说明数据量合理，但是不足一行，那么下次再处理
                    return true;
                }
                // 解析到单个换行符了
                if(line == "\n" || line == "\r\n")
                {
                    break;// 结束循环
                }
                bool ret = ParseHttpHead(line);
                if(ret == false)
                {
                    ParseERROR(400);
                    return false;
                }
            }
            // 头部信息处理完毕，进入正文获取阶段
            _recv_statu = RECV_HTTP_BODY;
            return true;
        }
        // 解析头部信息
        bool ParseHttpHead(std::string& line)
        {
            // key: val\r\n的格式
            //key: val\r\nkey: val\r\n....
            if(line.back() == '\n') line.pop_back(); // 如果末尾是换行，就去掉换行
            if(line.back() == '\r') line.pop_back(); // 如果末尾是回车，就去掉回车
            size_t pos = line.find(": ");
            if(pos == std::string::npos)
            {
                // 说明不符合格式
                ParseERROR(400); // BAD REQUEST
                return false;
            }
            // 这里就不用解码了
            std::string key = line.substr(0,pos);
            std::string val = line.substr(pos + 2);
            _request.SetHeader(key,val);
            return true;
        }
        // 获取正文信息
        bool RecvHttpBody(Buffer& buf)
        {
            if(_recv_statu != RECV_HTTP_BODY) return false;
            // 1.先获取正文长度
            size_t content_length  = _request.ContentLength();
            if(content_length  == 0)
            {
                // 没有正文，直接解析完毕
                _recv_statu = RECV_HTTP_OVER;
                return true;
            }   
            // 2.获取当前已经接收了多少正文长度，就是_request._body 中放了多少数据了，然后总长 - 已经接收 = 实际还需要接收多少
            size_t real_len = content_length - _request._body.size();
            // 3. 接收正文放到body中，但是也要考虑当前缓冲区中的数据，是否是全部的正文
            //  3.1 缓冲区中数据，包含了当前请求的所有正文，则取出所需的数据
            if(buf.ReadAbleSize() >= real_len)
            {
                _request._body.append(buf.ReadPosition(),real_len);
                buf.MoveReadOffset(real_len);
                _recv_statu = RECV_HTTP_OVER; // 解析完毕
                return true;
            }
            //  3.2 缓冲区中数据，无法满足当前正文的需要，数据不足，取出数据，然后等待新数据到来
            _request._body.append(buf.ReadPosition(),buf.ReadAbleSize());
            buf.MoveReadOffset(buf.ReadAbleSize());
            return true;
        }
    public:
        HttpContext() : _resp_statu(200), _recv_statu(RECV_HTTP_LINE) {}
        // 重置数据
        void ReSet()
        {
            _resp_statu = 200;
            _recv_statu = RECV_HTTP_LINE;
            _request.Reset();
        }
        // 返回响应状态码
        int RespStatu() { return _resp_statu; }
        // 返回当前的处理阶段
        HttpRecvStatu RecvStatu() { return _recv_statu; }
        // 返回当前响应的引用
        HttpRequest& Request() { return _request; }
        // 进行一次接收并处理http请求，可能会调用多次，因为请求的数据可能没有完全提取上来，需要下次再处理
        void RecvHttpRequest(Buffer& buf)
        {
            switch(_recv_statu)
            {
                // case后不需要break，因为顺利的话可以一次性执行完毕
                case RECV_HTTP_LINE:RecvHttpLine(buf);
                case RECV_HTTP_HEAD:RecvHttpHead(buf);
                case RECV_HTTP_BODY:RecvHttpBody(buf);
            }
            return;
        }
};

class HttpServer 
{
    private:
        using Handler = std::function<void(const HttpRequest&,HttpResponse*)>; // 路由表里的回调函数
        using Handlers = std::vector<std::pair<std::regex,Handler>>; // 路由表 其中first里面存的正则表达式的匹配规则
        Handlers _get_route;
        Handlers _post_route;
        Handlers _put_route;
        Handlers _delete_route;
        std::string _basedir; // 静态资源目录
        TcpServer _server;
    private:
        // 触发错误时调用的函数
        void ErrorHandler(const HttpRequest& req,HttpResponse* rsp)
        {
            //1. 组织一个错误展示页面
            std::string body;
            body += "<html>";
            body += "<head>";
            body += "<meta http-equiv='Content-Type' content='text/html;charset=utf-8'>";
            body += "</head>";
            body += "<body>";
            body += "<h1>";
            body += std::to_string(rsp->_statu);
            body += " ";
            body += Util::StatuDesc(rsp->_statu);
            body += "</h1>";
            body += "</body>";
            body += "</html>";
            //2. 将页面数据，当作响应正文，放入rsp中
            rsp->SetContent(body, "text/html");
        }
        //将HttpResponse中的要素按照http协议格式进行组织，发送
        void WriteResponse(const PtrConnection& conn,const HttpRequest& req,HttpResponse& rsp)
        {
            // 1.先完善头部字段
            if(req.Close() == true)
            {
                rsp.SetHeader("Connection","close"); // 短链接
            }
            else  
            {
                rsp.SetHeader("Connection","keep-alive"); // 长链接 
            }
            if(rsp._body.size() > 0 && rsp.HasHeader("Content-Length") == false)
            {
                // 正文里有数据，并且rsp还没有设置正文长度字段
                rsp.SetHeader("Content-Length",std::to_string(rsp._body.size()));
            }
            if(rsp._body.size() > 0 && rsp.HasHeader("Content-Type") == false)
            {
                // 有正文数据，并且没有设置正文数据的类型
                rsp.SetHeader("Content-Type","application/octet-stream");
            }
            if(rsp._redirect_flag == true)
            {
                // 需要设置重定向
                rsp.SetHeader("Location",rsp._redirect_url);
            }
            // 2.将rsp里的类容按照http的格式进行组织
            std::stringstream rsp_str;
            rsp_str << req._version << " " << std::to_string(rsp._statu)  << " " << Util::StatuDesc(rsp._statu) << "\r\n";
            for(auto& p : rsp._headers)
            {
                rsp_str << p.first << ": " << p.second << "\r\n";
            }
            rsp_str << "\r\n"; // 空行
            rsp_str << rsp._body; // 正文
            // 3.发送数据
            conn->Send(rsp_str.str().c_str(),rsp_str.str().size());
        }
        // 是否是静态资源的访问
        bool IsFileHandler(const HttpRequest& req)
        {
            // 1.必须设置了静态资源的根目录
            if(req._path.empty() == true)
            {
                return false;
            }
            // // 2. 请求方法，必须是GET / HEAD请求方法
            if(req._method != "GET" && req._method != "HEAD")
            {
                return false;
            }
            // 3.请求资源的路径必须是合法路径
            if(Util::ValidPath(req._path) == false)
            {
                return false;
            }
            // 4. 请求的资源必须存在,且是一个普通文件
            //    有一种请求比较特殊 -- 目录：/, /image/， 这种情况给后边默认追加一个 index.html
            // index.html    /image/a.png
            // 不要忘了前缀的相对根目录,也就是将请求路径转换为实际存在的路径  /image/a.png  ->   ./wwwroot/image/a.png
            std::string req_path = _basedir + req._path; // 避免直接修改req里面的值，所以用了临时对象
            if(req._path.back() == '/') // 说明默认访问首页
            {
                req_path += "index.html";
            }
            if(Util::IsRegular(req_path) == false)
            {
                return false;
            }
            return true;
        }
        //静态资源的请求处理 --- 将静态资源文件的数据读取出来，放到rsp的_body中, 并设置mime
        void FileHandler(const HttpRequest& req,HttpResponse* rsp)
        {
            std::string req_path = _basedir + req._path; // 避免直接修改req里面的值，所以用了临时对象
            if(req._path.back() == '/') // 说明默认访问首页
            {
                req_path += "index.html";
            }
            bool ret = Util::ReadFile(req_path,&rsp->_body);
            if(ret == false)
            {
                return;
            }
            std::string mime = Util::ExtMime(req_path);
            rsp->SetHeader("Content-Type",mime);
        }
        // 对功能性请求的分类处理
        void Dispatcher(HttpRequest& req,HttpResponse* rsp,Handlers& handlers)
        {
            //在对应请求方法的路由表中，查找是否含有对应资源请求的处理函数，有则调用，没有则调用404
            //思想：路由表存储的时键值对 -- 正则表达式 & 处理函数
            //使用正则表达式，对请求的资源路径进行正则匹配，匹配成功就调用对应函数进行处理
            // 比如：  /numbers/(\d+)       /numbers/12345
            for(auto& handler : handlers)
            {
                const std::regex re = handler.first;
                const Handler& functor = handler.second;
                bool ret = std::regex_match(req._path,req._matches,re);
                if(ret == false)
                {
                    continue;
                }
                return functor(req,rsp); // 传入req和空的rsp，执行处理函数
            }
            rsp->_statu = 404;
        }
        void Route(HttpRequest& req,HttpResponse* rsp)
        {
            //1. 对请求进行分辨，是一个静态资源请求，还是一个功能性请求
            //   静态资源请求，则进行静态资源的处理
            //   功能性请求，则需要通过几个请求路由表来确定是否有处理函数
            //   既不是静态资源请求，也没有设置对应的功能性请求处理函数，就返回405
            if(IsFileHandler(req) == true)
            {
                return FileHandler(req,rsp);
            }
            if(req._method == "GET" || req._method == "HEAD")
            {
                return Dispatcher(req,rsp,_get_route);
            }
            else if(req._method == "POST")
            {
                return Dispatcher(req,rsp,_post_route);
            }
            else if(req._method == "PUT")
            {
                return Dispatcher(req,rsp,_put_route);
            }
            else if(req._method == "DELETE")
            {
                return Dispatcher(req,rsp,_delete_route);
            }
            rsp->_statu = 405; // Method Not Allowed
        }
        // 设置上下文
        void OnConnected(const PtrConnection& conn)
        {
            conn->SetContext(HttpContext());
            DBG_LOG("New Connection %p",conn.get());
        }
        // 缓冲区的数据解析 + 处理
        void OnMessage(const PtrConnection& conn,Buffer& buf)
        {
            while(buf.ReadAbleSize() > 0)
            {
                // 1.先获取上下文
                HttpContext* context = conn->GetContext()->get<HttpContext>();
                //2. 通过上下文对缓冲区数据进行解析，得到HttpRequest对象
                //  1. 如果缓冲区的数据解析出错，就直接回复出错响应
                //  2. 如果解析正常，且请求已经获取完毕，才开始去进行处理
                context->RecvHttpRequest(buf);
                DBG_LOG("OnMessage-----------");
                HttpRequest& req = context->Request();
                DBG_LOG("OnMessage-----------");
                HttpResponse rsp(context->RespStatu());
                if(context->RespStatu() >= 400)
                {
                    // 出错了，直接组织错误信息然后返回关闭连接
                    ErrorHandler(req,&rsp); // 将错误信息填充到刚刚示例化的rsp对象中
                    WriteResponse(conn,req,rsp); // 组织错误响应，通过conn发送给对方
                    context->ReSet(); // 顺便重置这个连接的上下文
                    buf.MoveReadOffset(buf.ReadAbleSize()); // 把缓冲区信息全部清空
                    conn->Shutdown(); // 最后关闭连接
                    return;
                }
                if(context->RecvStatu() != RECV_HTTP_OVER)
                {
                    // 说明有数据未就绪，那么下次再处理
                    return;
                }
                // 3.请求路由 + 业务处理
                Route(req,&rsp);
                // 4.对HttpResponse进行组织发送
                WriteResponse(conn,req,rsp);
                // 5.重置上下文
                context->ReSet();
                // 6.根据是否是长短链接来关闭连接
                if(rsp.Close() == true) conn->Shutdown(); // 短连接直接关闭
            }
        }
    public:
        HttpServer(int port,int timeout = DEFALT_TIMEOUT) : _server(port)
        {
            _server.EnableInactiveRelease(timeout);
            _server.SetConnectedCallback(std::bind(&HttpServer::OnConnected,this,std::placeholders::_1));
            _server.SetMessageCallback(std::bind(&HttpServer::OnMessage, this, std::placeholders::_1, std::placeholders::_2));
        }
        // 设置静态资源目录
        void SetBaseDir(const std::string& path)
        {
            assert(Util::IsDirectory(path) == true);
            _basedir = path;
        }
        /*设置/添加，请求（请求的正则表达）与处理函数的映射关系*/
        void Get(const std::string &pattern, const Handler &handler)
        {
            _get_route.push_back({std::regex(pattern),handler});
        }
        void Post(const std::string &pattern, const Handler &handler)
        {
            _post_route.push_back({std::regex(pattern),handler});
        }
        void Put(const std::string &pattern, const Handler &handler)
        {
            _put_route.push_back({std::regex(pattern),handler});
        }
        void Delete(const std::string &pattern, const Handler &handler)
        {
            _delete_route.push_back({std::regex(pattern),handler});
        }
        void SetThreadCount(int count)
        {
            _server.SetThreadCount(count);
        }
        void Listen()
        {
            _server.Start();
        }
};